package spring.controller;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.Date;
import java.util.Iterator;

/**
 * project: forum
 * ate: 14-9-15
 * author: wzq
 * description: TODO
 */
@Controller
@RequestMapping("/postArticle")
//@Controller("/postArticle")
public class PostArticle {

    @RequestMapping(method = RequestMethod.GET)
    public ModelAndView test(HttpServletRequest req, HttpServletResponse response) throws IOException {
        System.out.println("/postArticle");
        return new ModelAndView("post_article_write");
    }

    /**
     * 通过流的方式上传文件
     * 将name=file控件得到的文件封装成CommonsMultipartFile 对象
     *
     * @RequestParam("file") 文件
     */
    @RequestMapping(value = "/fileUpload", method = RequestMethod.POST)
    @ResponseBody
    public String fileUpload(@RequestParam("file") CommonsMultipartFile file) throws IOException {
        System.out.println("/postArticle/fileUpload");
        //用来检测程序运行时间
        long startTime = System.currentTimeMillis();
        System.out.println("fileName：" + file.getOriginalFilename());

        try {
            //获取输出流
            OutputStream os = new FileOutputStream("/home/wzq/tmp/" + new Date().getTime() + "-" + file.getOriginalFilename());
            // OutputStream os = new FileOutputStream("E:/" + new Date().getTime() + file.getOriginalFilename());
            //获取输入流 CommonsMultipartFile 中可以直接得到文件的流
            InputStream is = file.getInputStream();
            int temp;
            //一个一个字节的读取并写入
            while ((temp = is.read()) != (-1)) {
                os.write(temp);
            }
            os.flush();
            os.close();
            is.close();

        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        long endTime = System.currentTimeMillis();
        System.out.println("方法一的运行时间：" + String.valueOf(endTime - startTime) + "ms");
        return "success";
    }

    /**
     * 采用file.Transto 来保存上传的文件
     */
    @RequestMapping(value = "fileUpload2", method = RequestMethod.POST)
    @ResponseBody
    public String fileUpload2(@RequestParam("file") CommonsMultipartFile file) throws IOException {
        System.out.println("/postArticle/fileUpload2");
        long startTime = System.currentTimeMillis();
        System.out.println("fileName：" + file.getOriginalFilename());
        String path = "/home/wzq/tmp/" + new Date().getTime() + "-" + file.getOriginalFilename();

        File newFile = new File(path);
        //通过CommonsMultipartFile的方法直接写文件（注意这个时候）
        file.transferTo(newFile);
        long endTime = System.currentTimeMillis();
        System.out.println("方法二的运行时间：" + String.valueOf(endTime - startTime) + "ms");
        return "success";
    }


    /**
     * 采用spring提供的上传文件的方法
     */
    @RequestMapping(value = "springUpload", method = RequestMethod.POST)
    @ResponseBody
    public String springUpload(HttpServletRequest request) throws IllegalStateException, IOException {
        System.out.println("/postArticle/springUpload");
        long startTime = System.currentTimeMillis();
        //将当前上下文初始化给  CommonsMutipartResolver （多部分解析器）
        CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(request.getSession().getServletContext());
        //检查form中是否有enctype="multipart/form-data"
        if (multipartResolver.isMultipart(request)) {
            //将request变成多部分request
            MultipartHttpServletRequest multiRequest = (MultipartHttpServletRequest) request;
            //获取multiRequest 中所有的文件名
            Iterator iter = multiRequest.getFileNames();
            while (iter.hasNext()) {
                //一次遍历所有文件
                MultipartFile file = multiRequest.getFile(iter.next().toString());
                if (file != null) {
                    String path = "/home/wzq/tmp/" + new Date().getTime() + "-" + file.getOriginalFilename();
                    //上传
                    file.transferTo(new File(path));
                }
            }
        }
        long endTime = System.currentTimeMillis();
        System.out.println("方法三的运行时间：" + String.valueOf(endTime - startTime) + "ms");
        return "success";
    }
}
